implementing data visualization in python